/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * See LICENSE.txt included in this distribution for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at LICENSE.txt.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright (c) 2021, Oracle and/or its affiliates. All rights reserved.
 */
package org.opengrok.indexer.util;

import org.jetbrains.annotations.NotNull;

import java.io.ObjectInputFilter;
import java.util.Set;

public class WhitelistObjectInputFilter implements ObjectInputFilter {

    private final Set<Class<?>> whitelist;

    public WhitelistObjectInputFilter(@NotNull final Class<?>... whitelist) {
        this.whitelist = Set.of(whitelist);
    }

    @Override
    public Status checkInput(final FilterInfo filterInfo) {
        if (filterInfo.serialClass() == null || whitelist.contains(filterInfo.serialClass())) {
            return Status.ALLOWED;
        }
        return Status.REJECTED;
    }
}
